import torch

# 检查 PyTorch 版本和 CUDA 支持
print(f"PyTorch 版本: {torch.__version__}")          # 示例输出: 2.3.0
print(f"CUDA 可用性: {torch.cuda.is_available()}")   # 输出 True 表示正常
print(f"CUDA 版本: {torch.version.cuda}")            # 示例输出: 11.8
